package com.suitsoft.ut.util;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;

import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.tools.zip.ZipOutputStream;

public class ZipManager {

 

        /**
         * zip压缩功能测试. 将d:\\temp\\zipout目录下的所有文件连同子目录压缩到d:\\temp\\out.zip.
         *
         * @param baseDir 所要压缩的目录名（包含绝对路径）
         * @param objFileName 压缩后的文件名
         * @throws Exception
         */
        public void createZip(String baseDir, String objFileName) throws Exception {
                File folderObject = new File(baseDir);

 

                if (folderObject.exists()){
                        List fileList = getSubFiles(new File(baseDir));

 

                        //压缩文件名
                        ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(objFileName));

 

                        ZipEntry ze = null;
                        byte[] buf = new byte[1024];
                        int readLen = 0;
                        for (int i = 0; i < fileList.size(); i++) {
                                File f = (File) fileList.get(i);
                                System.out.println("Adding: " + f.getPath() + f.getName());

 

                                //创建一个ZipEntry，并设置Name和其它的一些属性
                                ze = new ZipEntry(getAbsFileName(baseDir, f));
                                ze.setSize(f.length());
                                ze.setTime(f.lastModified());
 

                                //将ZipEntry加到zos中，再写入实际的文件内容
                                zos.putNextEntry(ze);
                                zos.setEncoding("gbk");
                                InputStream is = new BufferedInputStream(new FileInputStream(f));
                                while ((readLen = is.read(buf, 0, 1024)) != -1) {
                                        zos.write(buf, 0, readLen);
                                }
                                is.close();
                                System.out.println("done...");
                        }
                        zos.close();
                }else{
                        throw new Exception("this folder isnot exist!");
                }
        }
        /**
         * zip压缩功能测试. 将指定文件压缩后存到一压缩文件中
         *
         * @param baseDir 所要压缩的文件名
         * @param objFileName 压缩后的文件名
         * @return 压缩后文件的大小
         * @throws Exception
         */
        public long createFileToZip(String zipFilename,String sourceFileName) throws Exception {

 

                File  sourceFile = new File(sourceFileName);

 

                byte[] buf = new byte[1024];

 

                //压缩文件名
                File objFile = new File(zipFilename);

 

                ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(objFile));

 

                ZipEntry ze = null;
                //创建一个ZipEntry，并设置Name和其它的一些属性
                ze = new ZipEntry(sourceFile.getName());
                ze.setSize(sourceFile.length());
                ze.setTime(sourceFile.lastModified());

 

                //将ZipEntry加到zos中，再写入实际的文件内容
                zos.putNextEntry(ze);

 

                InputStream is = new BufferedInputStream(new FileInputStream(sourceFile));

 

                int readLen = -1;
                while ((readLen = is.read(buf, 0, 1024)) != -1) {
                        zos.write(buf, 0, readLen);
                }
                is.close();
                zos.close();

 


                return objFile.length();
        }
        /**
         * zip压缩功能测试. 将指定文件压缩后存到一压缩文件中
         *
         * @param baseDir 所要压缩的文件名
         * @param objFileName 压缩后的文件名
         * @return 压缩后文件的大小
         * @throws Exception
         */
        public long createFileToZip(File sourceFile,File zipFile)throws IOException {

 


                byte[] buf = new byte[1024];

 

                //压缩文件名
                File objFile = zipFile;

 

                ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(objFile));

 

                ZipEntry ze = null;
                //创建一个ZipEntry，并设置Name和其它的一些属性
                ze = new ZipEntry(sourceFile.getName());
                ze.setSize(sourceFile.length());
                ze.setTime(sourceFile.lastModified());

 

                //将ZipEntry加到zos中，再写入实际的文件内容
                zos.putNextEntry(ze);

 

                InputStream is = new BufferedInputStream(new FileInputStream(sourceFile));

 

                int readLen = -1;
                while ((readLen = is.read(buf, 0, 1024)) != -1) {
                        zos.write(buf, 0, readLen);
                }
                is.close();
                zos.close();

 

                return objFile.length();
        }

 

        /**
         * 测试解压缩功能. 将d:\\download\\test.zip连同子目录解压到d:\\temp\\zipout目录下.
         *
         * @throws Exception
         */
        public void releaseZipToFile(String sourceZip, String outFileName)
                        throws IOException{
                      ZipFile zfile=new ZipFile(sourceZip);
                      Enumeration zList=zfile.getEntries();
                      ZipEntry ze=null;
                      byte[] buf=new byte[1024];
                      while(zList.hasMoreElements()){
                      //从ZipFile中得到一个ZipEntry
                      ze=(ZipEntry)zList.nextElement();
                      if(ze.isDirectory()){
                      continue;
                      }
                      //以ZipEntry为参数得到一个InputStream，并写到OutputStream中
                      OutputStream os=new BufferedOutputStream(new FileOutputStream(getRealFileName(outFileName, ze.getName())));
                      InputStream is=new BufferedInputStream(zfile.getInputStream(ze));
                      int readLen=0;
                      while ((readLen=is.read(buf, 0, 1024))!=-1) {
                      os.write(buf, 0, readLen);
                      }
                      is.close();
                      os.close();
                      System.out.println("Extracted: "+ze.getName());
                      }
                      zfile.close();

        }

 

        /**
         * 取得指定目录下的所有文件列表，包括子目录.
         *
         * @param baseDir
         *            File 指定的目录
         * @return 包含java.io.File的List
         */
        private List getSubFiles(File baseDir) {
                List ret = new ArrayList();
                //File base=new File(baseDir);
                File[] tmp = baseDir.listFiles();
                for (int i = 0; i < tmp.length; i++) {
                        if (tmp[i].isFile()) {
                                ret.add(tmp[i]);
                        }
                        if (tmp[i].isDirectory()) {
                                ret.addAll(getSubFiles(tmp[i]));
                        }
                }
                return ret;
        }

 

        /**
         * 给定根目录，返回一个相对路径所对应的实际文件名.
         *
         * @param baseDir
         *            指定根目录
         * @param absFileName
         *            相对路径名，来自于ZipEntry中的name
         * @return java.io.File 实际的文件
         */
        private File getRealFileName(String baseDir, String absFileName) {
                String[] dirs = absFileName.split("/");
                //System.out.println(dirs.length);
                File ret = new File(baseDir);
                //System.out.println(ret);
                if (dirs.length > 1) {
                        for (int i = 0; i < dirs.length - 1; i++) {
                                ret = new File(ret, dirs[i]);
                        }
                }
                if (!ret.exists()) {
                        ret.mkdirs();
                }
                ret = new File(ret, dirs[dirs.length - 1]);
                return ret;
        }

 

        /**
         * 给定根目录，返回另一个文件名的相对路径，用于zip文件中的路径.
         *
         * @param baseDir
         *            java.lang.String 根目录
         * @param realFileName
         *            java.io.File 实际的文件名
         * @return 相对文件名
         */
        private String getAbsFileName(String baseDir, File realFileName) {
                File real = realFileName;
                File base = new File(baseDir);
                String ret = real.getName();
                while (true) {
                        real = real.getParentFile();
                        if (real == null)
                                break;
                        if (real.equals(base))
                                break;
                        else {
                                ret = real.getName() + "/" + ret;
                        }
                }
                System.out.println("TTTTT" + ret);
                return ret;
        }


        public void testReadZip() throws Exception{
        String baseDir="d:\\temp\\zipout";
        ZipFile zfile=new ZipFile("d:\\download\\src.zip");
        Enumeration zList=zfile.getEntries();
        ZipEntry ze=null;
        byte[] buf=new byte[1024];
        while(zList.hasMoreElements()){
        //从ZipFile中得到一个ZipEntry
        ze=(ZipEntry)zList.nextElement();
        if(ze.isDirectory()){
        continue;
        }
        //以ZipEntry为参数得到一个InputStream，并写到OutputStream中
        OutputStream os=new BufferedOutputStream(new FileOutputStream(getRealFileName(baseDir, ze.getName())));
        InputStream is=new BufferedInputStream(zfile.getInputStream(ze));
        int readLen=0;
        while ((readLen=is.read(buf, 0, 1024))!=-1) {
        os.write(buf, 0, readLen);
        }
        is.close();
        os.close();
        System.out.println("Extracted: "+ze.getName());
        }
        zfile.close();
        }


   public static void main(String args[]){
     ZipManager manager = new ZipManager();
     try {
    	 manager.createZip("F:\\workSpace\\.metadata\\.plugins\\org.eclipse.wst.server.core\\tmp1\\wtpwebapps\\ut\\WEB-INF\\exp\\temp\\1358127175597", "F:\\workSpace\\.metadata\\.plugins\\org.eclipse.wst.server.core\\tmp1\\wtpwebapps\\ut\\WEB-INF\\exp\\temp\\a.zip");
     }
     catch (Exception e) {}
     System.out.println("over");
   }
 


}

